# -*- coding: utf-8 -*-
"""
配置文件 - 数据库连接和API设置
"""

# PostgreSQL 数据库配置
DATABASE_CONFIG = {
    'host': '192.168.110.10',        # 数据库主机地址
    'port': 5432,              # 数据库端口
    'database': 'postgres', # 数据库名称
    'user': 'postgres',        # 用户名
    'password': ''     # 密码
}

# Ollama 本地模型配置
OLLAMA_CONFIG = {
    'base_url': 'http://192.168.110.10:11434',  # Ollama服务地址
    'embedding_model': 'modelscope.cn/qwen/Qwen3-Embedding-0.6B-GGUF:latest', # 嵌入模型
    'chat_model': 'modelscope.cn/qwen/Qwen3-32B-GGUF:latest'            # 聊天模型
}

# 远程API配置 (备选方案)
REMOTE_API_CONFIG = {
    'api_key': 'your_api_key_here',
    'base_url': 'https://api.openai.com/v1',
    'embedding_model': 'text-embedding-ada-002',
    'chat_model': 'gpt-3.5-turbo'
}

# 向量维度配置
VECTOR_DIMENSION = 1024  # nomic-embed-text的向量维度

# 表名配置
TABLE_NAME = 'ly'
